#' Train cell type classifier
#'
#' @description Train a classifier for a new cell type.
#' If cell type has a parent, only available for \code{\link{scAnnotatR}}
#' object as parent cell classifying model.
#'
#' @param train_obj object that can be used for training the new model.
#' \code{\link{Seurat}} object or \code{\link{SingleCellExperiment}} object
#' is supported.
#' If the training model has parent, parent_tag_slot may have been indicated.
#' This field would have been filled out automatically
#' if user precedently run classify_cells function.
#' If no (predicted) cell type annotation provided,
#' the function can be run if 1- parent_cell or 2- parent_classifier is provided.
#' @param assay name of assay to use in training object.
#' @param slot type of expression data to use in training object, omitted if
#' train_obj is \code{\link{SingleCellExperiment}} object.
#' @param cell_type string indicating the name of the subtype
#' This must exactly match cell tag/label if cell tag/label is a string.
#' @param marker_genes list of marker genes used for the new training model
#' @param tag_slot string, name of slot in cell meta data
#' indicating cell tag/label in the training object.
#' Strings indicating cell types are expected in this slot.
#' For \code{\link{Seurat}} object, default value is "active.ident".
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param parent_cell string indicated the name of the parent cell type,
#' if parent cell type classifier has already been saved in model database.
#' Adjust path_to_models for exact database.
#' @param parent_tag_slot string, name of a slot in cell meta data
#' indicating assigned/predicted cell type. Default is "predicted_cell_type".
#' This slot would have been filled automatically
#' if user have called classify_cells function.
#' The slot must contain only string values.
#' @param parent_classifier classification model for the parent cell type
#' @param path_to_models path to the folder containing the model database.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param zscore whether gene expression in train_obj is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return \code{\link{scAnnotatR}} object
#'
#' @note Only one cell type is expected for each cell in object.
#' Ambiguous cell type, such as: "T cells/NK cells/ILC",
#' will be ignored from training.
#' Subtypes used in training model for parent cell types must be indicated
#' as parent cell type. For example, when training for B cells,
#' plasma cells must be annotated as B cells in order to be used.
#'
#' @examples
#' # load small example dataset
#' data("tirosh_mel80_example")
#'
#' # this dataset already contains pre-defined cell labels
#' table(Seurat::Idents(tirosh_mel80_example))
#'
#' # define genes to use to classify this cell type (B cells in this example)
#' selected_marker_genes_B = c("CD19", "MS4A1", "CD79A")
#'
#' # train the classifier, the "cell_type" argument must match
#' # the cell labels in the data, except upper/lower case
#' set.seed(123)
#' classifier_b <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', marker_genes = selected_marker_genes_B,
#' cell_type = "b cells", tag_slot = 'active.ident')
#'
#' # classify cell types using B cell classifier,
#' # a test classifier process may be used before applying the classifier
#' tirosh_mel80_example <- classify_cells(classify_obj = tirosh_mel80_example,
#' classifiers = c(classifier_b), assay = 'RNA', slot = 'counts')
#'
#' # tag all cells that are plasma cells (random example here)
#' tirosh_mel80_example[['plasma_cell_tag']] <- c(rep(1, 80), rep(0, 400))
#'
#' # set new marker genes for the subtype
#' p_marker_genes = c("SDC1", "CD19", "CD79A")
#'
#' # train the classifier, the "B cell" classifier is used as parent.
#' # This means, only cells already classified as "B cells" will be evaluated.
#' # the "tag_slot" parameter tells the classifier to use this cell meta data
#' # for the training process.
#' set.seed(123)
#' plasma_classifier <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', cell_type = 'Plasma cell',
#' marker_genes = p_marker_genes, tag_slot = 'plasma_cell_tag',
#' parent_classifier = classifier_b)
#'
#' @export
train_classifier <- function(train_obj, assay, slot = NULL,
                             cell_type, marker_genes, tag_slot,
                             parent_cell = NA_character_,
                             parent_tag_slot = 'predicted_cell_type',
                             parent_classifier = NULL, path_to_models = "default",
                             zscore = TRUE, ambiguous_chars = NULL) {
  if (is(train_obj, 'Seurat')) {
    object <-
      train_classifier_seurat(train_obj, cell_type, marker_genes,
                              parent_cell, parent_classifier, path_to_models,
                              zscore, tag_slot, parent_tag_slot, assay, slot,
                              ambiguous_chars)
  } else if (is(train_obj, 'SingleCellExperiment')) {
    object <-
      train_classifier_sce(train_obj, cell_type, marker_genes,
                              parent_cell, parent_classifier, path_to_models,
                              zscore, tag_slot, parent_tag_slot, assay,
                              ambiguous_chars)
  } else {
    stop('Training object of not supported class', call. = FALSE)
  }

  return(object)
}

#' Train cell type classifier, when train_obj is Seurat object
#'
#' @description Train a classifier for a new cell type
#' If cell type has a parent, only available for \code{\link{scAnnotatR}}
#' object as parent cell classifying model.
#'
#' @param train_obj Seurat object
#' @param seurat_assay name of assay to use in training object.
#' @param seurat_slot type of expression data to use in training object
#' @param cell_type string indicating the name of the subtype
#' This must exactly match cell tag/label if cell tag/label is a string.
#' @param marker_genes list of marker genes used for the new training model
#' @param seurat_tag_slot string, name of slot in cell meta data
#' indicating cell tag/label in the training object.
#' Strings indicating cell types are expected in this slot.
#' For \code{\link{Seurat}} object, default value is "active.ident".
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param parent_cell string indicated the name of the parent cell type,
#' if parent cell type classifier has already been saved in model database.
#' Adjust path_to_models for exact database.
#' @param seurat_parent_tag_slot string, name of a slot in cell meta data
#' indicating assigned/predicted cell type. Default is "predicted_cell_type".
#' This slot would have been filled automatically
#' if user have called classify_cells function.
#' The slot must contain only string values.
#' @param parent_classifier classification model for the parent cell type
#' @param path_to_models path to the folder containing the model database.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param zscore whether gene expression in train_obj is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return \code{\link{scAnnotatR}} object
#'
#' @rdname internal
train_classifier_seurat <-
  function(train_obj, cell_type, marker_genes, parent_cell = NA_character_,
           parent_classifier = NULL, path_to_models = "default", zscore = TRUE,
           seurat_tag_slot, seurat_parent_tag_slot = 'predicted_cell_type',
           seurat_assay, seurat_slot, ambiguous_chars) {

  preprocessed <- preprocess_seurat_object(train_obj, seurat_assay, seurat_slot,
                                           seurat_tag_slot, seurat_parent_tag_slot)

  object <- train_classifier_from_mat(preprocessed$mat, preprocessed$tag,
                                      cell_type, marker_genes,
                                      preprocessed$parent_tag, parent_cell,
                                      parent_classifier, path_to_models, zscore,
                                      ambiguous_chars)
  return(object)
}

#' Train cell type classifier, when train_obj is SCE object
#'
#' @description Train a classifier for a new cell type
#' If cell type has a parent, only available for \code{\link{scAnnotatR}}
#' object as parent cell classifying model.
#'
#' @param train_obj SCE object
#' @param sce_assay name of assay to use in training object.
#' @param cell_type string indicating the name of the subtype
#' This must exactly match cell tag/label if cell tag/label is a string.
#' @param marker_genes list of marker genes used for the new training model
#' @param sce_tag_slot string, name of slot in cell meta data
#' indicating cell tag/label in the training object.
#' Strings indicating cell types are expected in this slot.
#' For \code{\link{Seurat}} object, default value is "active.ident".
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param parent_cell string indicated the name of the parent cell type,
#' if parent cell type classifier has already been saved in model database.
#' Adjust path_to_models for exact database.
#' @param sce_parent_tag_slot string, name of a slot in cell meta data
#' indicating assigned/predicted cell type. Default is "predicted_cell_type".
#' This slot would have been filled automatically
#' if user have called classify_cells function.
#' The slot must contain only string values.
#' @param parent_classifier classification model for the parent cell type
#' @param path_to_models path to the folder containing the model database.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param zscore whether gene expression in train_obj is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return \code{\link{scAnnotatR}} object
#'
#' @rdname internal
train_classifier_sce <-
  function(train_obj, cell_type, marker_genes, parent_cell = NA_character_,
           parent_classifier = NULL, path_to_models = "default", zscore = TRUE,
           sce_tag_slot, sce_parent_tag_slot = "predicted_cell_type", sce_assay,
           ambiguous_chars = NULL) {

  preprocessed <- preprocess_sce_object(train_obj, sce_assay, sce_tag_slot,
                                        sce_parent_tag_slot)
  object <- train_classifier_from_mat(preprocessed$mat, preprocessed$tag,
                                      cell_type, marker_genes,
                                      preprocessed$parent_tag, parent_cell,
                                      parent_classifier, path_to_models, zscore,
                                      ambiguous_chars)

  return(object)
}

#' Train cell type from matrix
#'
#' @description Train a classifier for a new cell type from expression matrix
#' and tag
#' If cell type has a parent, only available for \code{\link{scAnnotatR}}
#' object as parent cell classifying model.
#' @param mat expression matrix of size n x m, n: genes, m: cells
#' @param tag named list indicating cell label
#' @param cell_type string indicating the name of the subtype
#' This must exactly match cell tag/label if cell tag/label is a string.
#' @param parent_tag named list indicating parent cell type
#' @param marker_genes list of marker genes used for the new training model
#' @param parent_cell string indicated the name of the parent cell type,
#' if parent cell type classifier has already been saved in model database.
#' Adjust path_to_models for exact database.
#' @param parent_classifier classification model for the parent cell type
#' @param path_to_models path to the folder containing the model database.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param zscore whether gene expression in train_obj is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return caret trained model
#'
#' @rdname internal
train_classifier_from_mat <- function(mat, tag, cell_type, marker_genes,
                                  parent_tag, parent_cell, parent_classifier,
                                  path_to_models, zscore,
                                  ambiguous_chars = NULL) {
  #--- part of parent cell type
  processed_parent <- process_parent_classifier(
    mat, parent_tag, parent_cell, parent_classifier, path_to_models, zscore
  )

  # check parent-child coherence
  if (!is.null(processed_parent$pos_parent)) {
    tag <- check_parent_child_coherence(
      mat, tag, processed_parent$pos_parent, processed_parent$parent_cell,
      cell_type, cell_type)
  }
  #--- end of part of parent cell type

  #--- part of cell type
  # filter cells
  filt <- filter_cells(mat, tag, ambiguous_chars)
  train_mat <- filt$mat
  train_tag <- filt$tag

  # marker genes selection
  train_mat <- select_marker_genes(train_mat, marker_genes)

  # transpose mat
  train_mat <- t(as.matrix(train_mat))

  # transform mat to zscore values
  if (zscore == TRUE)
    train_mat <- transform_to_zscore(train_mat)

  # construct cell tag to yes/no values
  train_tag <- construct_tag_vect(train_tag, cell_type) # should be named list

  # if cell type not found in tag
  if (all(train_tag != "yes")) {
    stop("Cell type not available in train data. Please verify cell type.",
         call. = FALSE)
  }

  # transform list to factor
  train_tag <- factor(train_tag, levels = c('yes', 'no'))

  # convert hyphen (-) by underscore (_)
  colnames(train_mat) <- gsub('-', '_', colnames(train_mat))
  # add G_ to beginning of gene names to prevent starting by digits
  colnames(train_mat) <- unlist(lapply(colnames(train_mat),
                                function(x) if(grepl('^[[:digit:]]', x))
                                  {paste0('G_', x)} else {x}))

  # train
  caret_model <- train_func(train_mat, train_tag)

  # remove this info to reduce memory
  caret_model$resampledCM <- caret_model$call <- caret_model$times <- NULL
  p_thres <- 0.5

  marker_genes <- labels(caret_model$terms)
  marker_genes <- gsub('^G_', '', marker_genes) # convert back underscore to hyphen
  marker_genes <- gsub('_', '-', marker_genes)
  object <- scAnnotatR(cell_type, caret_model, marker_genes, p_thres,
                       NA_character_)

  # only assign parent if pretrained model for parent cell type is avai
  parent_check <-
    (
      !is.null(processed_parent$parent.classifier) &&
        tolower(cell_type(processed_parent$parent.classifier)) ==
        tolower(processed_parent$parent_cell)
    ) || (
      tolower(processed_parent$parent_cell) %in%
        tolower(names(processed_parent$model_list))
    )
  if (parent_check) parent(object) <- processed_parent$parent_cell

  return(object)
}

#' Preprocess Seurat object
#'
#' @description Preprocess Seurat object to produce expression matrix,
#' tag, parent cell tag.
#'
#' @param seurat_obj Seurat object
#' @param seurat_assay name of assay to use in training object.
#' @param seurat_slot type of expression data to use in training object
#' @param seurat_tag_slot string, name of slot in cell meta data
#' indicating cell tag/label in the training object.
#' Strings indicating cell types are expected in this slot.
#' For \code{\link{Seurat}} object, default value is "active.ident".
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param seurat_parent_tag_slot string, name of a slot in cell meta data
#' indicating assigned/predicted cell type. Default is "predicted_cell_type".
#' This slot would have been filled automatically
#' if user have called classify_cells function.
#' The slot must contain only string values.
#' @return a list containing: expression matrix of size n x m, n: genes, m: cells;
#' a vector indicating cell type, and a vector containing parent cell type.
#'
#' @importFrom Seurat GetAssayData Idents
#'
#' @rdname internal
preprocess_seurat_object <- function(seurat_obj, seurat_assay, seurat_slot,
                                     seurat_tag_slot, seurat_parent_tag_slot) {
  # convert Seurat object to matrix
  mat = Seurat::GetAssayData(object = seurat_obj,
                             assay = seurat_assay, slot = seurat_slot)

  if (seurat_tag_slot == "active.ident") {
    tag <- Seurat::Idents(seurat_obj)
  } else {
    tag <- seurat_obj[[seurat_tag_slot]][,1]
    names(tag) <- colnames(seurat_obj)
  }

  if (seurat_parent_tag_slot == "active.ident") {
    parent_tag <- Seurat::Idents(seurat_obj)
  } else if (seurat_parent_tag_slot %in% colnames(seurat_obj[[]])) {
    parent_tag <- seurat_obj[[seurat_parent_tag_slot]][,1]
    names(parent_tag) <- colnames(seurat_obj)
  } else parent_tag <- NULL

  return_val <- list('mat' = mat, 'tag' = tag, 'parent_tag' = parent_tag)
  return(return_val)
}

#' Preprocess Seurat object
#'
#' @description Preprocess Seurat object to produce expression matrix,
#' tag, parent cell tag.
#'
#' @param sce_obj Seurat object
#' @param sce_assay name of assay to use in training object.
#' @param sce_tag_slot string, name of slot in cell meta data
#' indicating cell tag/label in the training object.
#' Strings indicating cell types are expected in this slot.
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param sce_parent_tag_slot string, name of a slot in cell meta data
#' indicating assigned/predicted cell type. Default is "predicted_cell_type".
#' This slot would have been filled automatically
#' if user have called classify_cells function.
#' The slot must contain only string values.
#' @return a list containing: expression matrix of size n x m, n: genes, m: cells;
#' a vector indicating cell type, and a vector containing parent cell type.
#'
#' @import SingleCellExperiment
#' @importFrom SummarizedExperiment assay
#'
#' @rdname internal
preprocess_sce_object <- function(sce_obj, sce_assay, sce_tag_slot,
                                  sce_parent_tag_slot) {
  # solve duplication of cell names
  colnames(sce_obj) <- make.unique(colnames(sce_obj), sep = '_')

  # convert SCE object to matrix
  mat = SummarizedExperiment::assay(sce_obj, sce_assay)

  tag = SummarizedExperiment::colData(sce_obj)[, sce_tag_slot]
  names(tag) <- colnames(sce_obj)

  if (sce_parent_tag_slot %in% colnames(SummarizedExperiment::colData(sce_obj))) {
    parent_tag <- SummarizedExperiment::colData(sce_obj)[, sce_parent_tag_slot]
    names(parent_tag) <- colnames(sce_obj)
  } else parent_tag <- NULL

  return_val <- list('mat' = mat, 'tag' = tag, 'parent_tag' = parent_tag)

  return(return_val)
}

#' Testing process.
#'
#' @description Testing process.
#'
#' @param test_obj object that can be used for testing
#' @param assay name of assay to use in test_object
#' @param slot type of expression data to use in test_object.
#' For Seurat object, some available types are: "counts", "data" and "scale.data".
#' Ignore this if test_obj is \code{\link{SingleCellExperiment}} object.
#' @param classifier scAnnotatR classification model
#' @param tag_slot string, name of annotation slot
#' indicating cell tag/label in the testing object.
#' Strings indicating cell types are expected in this slot.
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param target_cell_type vector indicating other cell types than cell labels
#' that can be considered as the main cell type in classifier,
#' for example, c("plasma cell", "b cell", "b cells", "activating b cell").
#' Default as NULL.
#' @param parent_classifier \code{\link{scAnnotatR}} object
#' corresponding to classification model for the parent cell type
#' @param parent_tag_slot string, name of tag slot in cell meta data
#' indicating pre-assigned/predicted parent cell type.
#' Default field is "predicted_cell_type".
#' The slot must contain only string values.
#' @param path_to_models path to the folder containing the list of models.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing
#' the new_models.rda file.
#' @param zscore boolean, whether gene expression is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return result of testing process in form of a list,
#' including predicted values, prediction accuracy at a probability threshold,
#' and roc curve information.
#'
#' @note Only one cell type is expected for each cell.
#' Ambiguous cell type, such as: "T cells/NK cells/ILC", will be ignored.
#' Subtypes used in testing model for parent cell types can be indicated
#' as parent cell type, or can be indicated in target_cell_type.
#' For example, when testing for B cells, plasma cells can be annotated as
#' B cells, or target_cell_type is set c("plasma cells").
#'
#' @examples
#' # load small example dataset
#' data("tirosh_mel80_example")
#'
#' # train the classifier
#' selected_marker_genes_B = c("CD19", "MS4A1", "CD79A")
#' set.seed(123)
#' classifier_b <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', marker_genes = selected_marker_genes_B,
#' cell_type = "b cells", tag_slot = 'active.ident')
#'
#' # test the classifier, target cell type can be in other formats or
#' # alternative cell type that can be considered as the classified cell type
#' classifier_b_test <- test_classifier(classifier = classifier_b,
#' test_obj = tirosh_mel80_example, assay = 'RNA', slot = 'counts',
#' tag_slot = 'active.ident', target_cell_type = c("B cell"))
#' classifier_b_test
#'
#' @export
setGeneric("test_classifier",
           function(classifier, test_obj, assay, slot = NULL, tag_slot,
                    target_cell_type = NULL, parent_classifier = NULL,
                    parent_tag_slot = 'predicted_cell_type',
                    path_to_models = "default", zscore = TRUE,
                    ambiguous_chars = NULL)
  standardGeneric("test_classifier"))

#' @inherit test_classifier
#'
#' @rdname test_classifier
setMethod('test_classifier', c('classifier' = 'scAnnotatR'),
          function(classifier, test_obj, assay, slot = NULL, tag_slot,
                   target_cell_type = NULL, parent_classifier = NULL,
                   parent_tag_slot = 'predicted_cell_type',
                   path_to_models = "default", zscore = TRUE,
                   ambiguous_chars = NULL) {
  if (is(test_obj, 'Seurat')) {
    return_val <-
      test_classifier_seurat(test_obj, classifier, target_cell_type,
                             parent_classifier, path_to_models, zscore,
                             tag_slot, parent_tag_slot, assay, slot,
                             ambiguous_chars)
  } else if (is(test_obj, 'SingleCellExperiment')) {
    return_val <-
      test_classifier_sce(test_obj, classifier, target_cell_type,
                          parent_classifier, path_to_models, zscore,
                          tag_slot, parent_tag_slot, assay, ambiguous_chars)
  } else {
    stop('Testing object of not supported class', call. = FALSE)
  }
  return(return_val)
})

#' Testing process for Seurat object
#'
#' @description Testing process when test object is of type Seurat
#'
#' @param test_obj Seurat object used for testing
#' @param seurat_assay name of assay to use in test_object
#' @param seurat_slot type of expression data to use in test_object.
#' For Seurat object, some available types are: "counts", "data" and "scale.data".
#' @param classifier scAnnotatR classification model
#' @param seurat_tag_slot string, name of annotation slot
#' indicating cell tag/label in the testing object.
#' Strings indicating cell types are expected in this slot.
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param target_cell_type vector indicating other cell types than cell labels
#' that can be considered as the main cell type in classifier,
#' for example, c("plasma cell", "b cell", "b cells", "activating b cell").
#' Default as NULL.
#' @param parent_classifier \code{\link{scAnnotatR}} object
#' corresponding to classification model for the parent cell type
#' @param seurat_parent_tag_slot string, name of tag slot in cell meta data
#' indicating pre-assigned/predicted parent cell type.
#' Default field is "predicted_cell_type".
#' The slot must contain only string values.
#' @param path_to_models path to the folder containing the list of models.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing
#' the new_models.rda file.
#' @param zscore boolean, whether gene expression is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return result of testing process in form of a list,
#' including predicted values, prediction accuracy at a probability threshold,
#' and roc curve information.
#'
#' @rdname internal
test_classifier_seurat <-
  function(test_obj, classifier, target_cell_type = NULL,
           parent_classifier = NULL, path_to_models = "default", zscore = TRUE,
           seurat_tag_slot, seurat_parent_tag_slot = "predicted_cell_type",
           seurat_assay, seurat_slot, ambiguous_chars = NULL) {
  preprocessed <- preprocess_seurat_object(test_obj, seurat_assay, seurat_slot,
                                           seurat_tag_slot, seurat_parent_tag_slot)

  return_val <- test_classifier_from_mat(preprocessed$mat, preprocessed$tag,
                                         classifier, preprocessed$parent_tag,
                                         target_cell_type, parent_classifier,
                                         path_to_models, zscore, ambiguous_chars)
  return(return_val)
}

#' Testing process for SCE object
#'
#' @description Testing process when test object is of type SCE
#'
#' @param test_obj SCE object used for testing
#' @param sce_assay name of assay to use in test_object
#' @param classifier scAnnotatR classification model
#' @param sce_tag_slot string, name of annotation slot
#' indicating cell tag/label in the testing object.
#' Strings indicating cell types are expected in this slot.
#' Expected values are string (A-Z, a-z, 0-9, no special character accepted)
#' or binary/logical, 0/"no"/F/FALSE: not being new cell type,
#' 1/"yes"/T/TRUE: being new cell type.
#' @param target_cell_type vector indicating other cell types than cell labels
#' that can be considered as the main cell type in classifier,
#' for example, c("plasma cell", "b cell", "b cells", "activating b cell").
#' Default as NULL.
#' @param parent_classifier \code{\link{scAnnotatR}} object
#' corresponding to classification model for the parent cell type
#' @param sce_parent_tag_slot string, name of tag slot in cell meta data
#' indicating pre-assigned/predicted parent cell type.
#' Default field is "predicted_cell_type".
#' The slot must contain only string values.
#' @param path_to_models path to the folder containing the list of models.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing
#' the new_models.rda file.
#' @param zscore boolean, whether gene expression is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return result of testing process in form of a list,
#' including predicted values, prediction accuracy at a probability threshold,
#' and roc curve information.
#'
#' @rdname internal
test_classifier_sce <-
  function(test_obj, classifier, target_cell_type = NULL,
           parent_classifier = NULL, path_to_models = "default", zscore = TRUE,
           sce_tag_slot, sce_parent_tag_slot = "predicted_cell_type", sce_assay,
           ambiguous_chars = NULL) {

  preprocessed <- preprocess_sce_object(test_obj, sce_assay, sce_tag_slot,
                                        sce_parent_tag_slot)

  return_val <- test_classifier_from_mat(preprocessed$mat, preprocessed$tag,
                                         classifier, preprocessed$parent_tag,
                                         target_cell_type, parent_classifier,
                                         path_to_models, zscore, ambiguous_chars)

  return(return_val)
}

#' Run testing process from matrix and tag
#'
#' @description Testing process from matrix and tag
#' @param mat expression matrix of size n x m, n: genes, m: cells
#' @param tag named list indicating cell label
#' @param classifier classification model
#' @param parent_tag named list indicating parent cell type
#' @param target_cell_type vector indicating other cell types than cell labels
#' that can be considered as the main cell type in classifier,
#' for example, c("plasma cell", "b cell", "b cells", "activating b cell").
#' Default as NULL.
#' @param parent_classifier classification model for the parent cell type
#' @param path_to_models path to the folder containing the model database.
#' As default, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param zscore whether gene expression in train_obj is transformed to zscore
#' @param ambiguous_chars List of characters to indicate ambiguous cells. For
#'                        more details see \code{\link{filter_cells}}.
#'
#' @return model performance statistics
#'
#' @rdname internal
test_classifier_from_mat <- function(mat, tag, classifier, parent_tag,
                                 target_cell_type, parent_classifier,
                                 path_to_models, zscore,
                                 ambiguous_chars = NULL) {
  # target_cell_type check
  if (!tolower(cell_type(classifier)) %in% tolower(target_cell_type)) {
    target_cell_type <- append(target_cell_type, cell_type(classifier))
  }

  #--- parent cell type
  # process parent classifier
  processed_parent <- process_parent_classifier(
    mat, parent_tag, parent(classifier), parent_classifier, path_to_models,
    zscore)

  # check parent-child coherence
  if (!is.null(processed_parent$pos_parent)) {
    tag <- check_parent_child_coherence(
      mat, tag, processed_parent$pos_parent, parent(classifier),
      cell_type(classifier), target_cell_type)
  }

  #--- children cell type
  # filter cells
  filt <- filter_cells(mat, tag, ambiguous_chars)
  test_mat <- filt$mat
  test_tag <- filt$tag

  # perform marker genes selection
  test_mat <- select_marker_genes(test_mat, marker_genes(classifier))

  # transpose mat
  test_mat <- t(as.matrix(test_mat))

  # transform mat to zscore values
  if (zscore == TRUE) test_mat <- transform_to_zscore(test_mat)

  # construct cell tag to yes/no values
  test_tag <- construct_tag_vect(test_tag, target_cell_type)

  # if cell type not found in tag
  if (all(test_tag != "yes")) {
    stop("Cell type ", cell_type(classifier),
         " is not available in the test data.
    Overwrite the target cell type using the target_cell_type parameter
    or verify that you chose the correct test dataset.",
         call. = FALSE)
  }

  return_val <- test_performance(test_mat, classifier, test_tag)
  return(return_val)
}

#' Plot roc curve
#'
#' @param test_result result of test_classifier function
#'
#' @return ggplot2 roc curve
#' @examples
#' # load small example dataset
#' data("tirosh_mel80_example")
#'
#' # train a classifier, for ex: B cell
#' selected_marker_genes_B = c("CD19", "MS4A1", "CD79A")
#' set.seed(123)
#' classifier_b <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', marker_genes = selected_marker_genes_B,
#' cell_type = "b cells", tag_slot = 'active.ident')
#'
#' classifier_b_test <- test_classifier(classifier = classifier_b,
#' test_obj = tirosh_mel80_example, assay = 'RNA', slot = 'counts',
#' tag_slot = 'active.ident', target_cell_type = c("B cell"))
#'
#' # run plot curve on the test result
#' roc_curve <- plot_roc_curve(test_result = classifier_b_test)
#' @import ROCR
#' @import ggplot2
#' @export
plot_roc_curve <- function(test_result) {
  fpr <- tpr <- NULL

  data <- rbind(c(0, 0), test_result$overall_roc[, c(2:3)], c(1, 1))
  colnames(data) <- c('fpr', 'tpr')

  # plot ROC curve
  q <- ggplot2::ggplot(data=data.frame(data), aes(x=fpr, y=tpr))
  q <- q + ggplot2::geom_line()
  q <- q + ggplot2::xlab("False Positive Rate (1-Specificity)")
  q <- q + ggplot2::ylab("True Positive Rate (Sensitivity)")
  q
  return(q)
}

#' Classify cells from multiple models
#'
#' @param classify_obj the object containing cells to be classified
#' @param assay name of assay to use in classify_object
#' @param slot type of expression data to use in classify_object.
#' For Seurat object, some available types are:
#' "counts", "data" and "scale.data".
#' @param classifiers list of classification models.
#' The model is obtained from train_classifier function or available in current
#' working space.
#' Users may test the model using test_classifier before using this function.
#' If classifiers contain classifiers for sub cell types, classifiers for
#' parent cell type must be indicated first in order to be applied before
#' children classifiers.
#' If classifiers is NULL, the method will use all classifiers in database.
#' @param cell_types list of cell types containing models to be used
#' for classification, only applicable if the models have been saved to package.
#' @param path_to_models path to the folder containing the list of models.
#' As default value, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param chunk_size size of data chunks to be predicted separately.
#' This option is recommended for large datasets to reduce running time.
#' Default value at 5000, because smaller datasets can be predicted rapidly.
#' @param ignore_ambiguous_result return all ambiguous predictions
#' (multiple cell types) to empty
#' When this parameter turns to TRUE,
#' most probably predicted cell types will be ignored.
#' @param cluster_slot name of slot in meta data containing cluster
#' information, in case users want to have additional cluster-level
#' prediction
#'
#' @return the input object with new slots in cells meta data
#' New slots are: predicted_cell_type, most_probable_cell_type,
#' slots in form of [cell_type]_p, [cell_type]_class, and clust_pred
#' (if cluster_slot was provided).
#'
#' @examples
#' # load small example dataset
#' data("tirosh_mel80_example")
#'
#' # train one classifier for one cell type, for ex, B cell
#' # define genes to use to classify this cell type
#' selected_marker_genes_B = c("CD19", "MS4A1", "CD79A")
#'
#' # train the classifier
#' set.seed(123)
#' classifier_b <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', marker_genes = selected_marker_genes_B,
#' cell_type = "b cells", tag_slot = 'active.ident')
#'
#' # do the same thing with other cell types, for example, T cells
#' selected_marker_genes_T = c("CD4", "CD8A", "CD8B")
#' set.seed(123)
#' classifier_t <- train_classifier(train_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', marker_genes = selected_marker_genes_T,
#' cell_type = "T cells", tag_slot = 'active.ident')
#'
#' # create a list of classifiers
#' classifier_ls <- list(classifier_b, classifier_t)
#'
#' # classify cells with list of classifiers
#' seurat.obj <- classify_cells(classify_obj = tirosh_mel80_example,
#' assay = 'RNA', slot = 'counts', classifiers = classifier_ls)
#'
#' @export
classify_cells <- function(classify_obj, assay, slot = NULL, classifiers = NULL,
                           cell_types = "all", chunk_size = 5000,
                           path_to_models = "default",
                           ignore_ambiguous_result = FALSE,
                           cluster_slot = 'clusters') {
  if (is(classify_obj, 'Seurat')) {
    classify_obj <- classify_cells_seurat(classify_obj, classifiers, cell_types,
                                          chunk_size, path_to_models,
                                          ignore_ambiguous_result, cluster_slot,
                                          assay, slot)
  } else if (is(classify_obj, 'SingleCellExperiment')) {
    classify_obj <- classify_cells_sce(classify_obj, classifiers, cell_types,
                                       chunk_size, path_to_models,
                                       ignore_ambiguous_result, assay, cluster_slot)
  } else
    stop('Classified object of not supported class', call. = FALSE)

  return(classify_obj)
}

#' Classify cells from multiple models for Seurat object as input
#'
#' @param classify_obj the Seurat object containing cells to be classified
#' @param seurat_assay name of assay to use in Seurat object
#' @param seurat_slot type of expression data to use in Seurat object.
#' Some available types are: "counts", "data" and "scale.data".
#' @param classifiers list of classification models.
#' The model is obtained from train_classifier function or available in current
#' working space.
#' Users may test the model using test_classifier before using this function.
#' If classifiers contain classifiers for sub cell types, classifiers for
#' parent cell type must be indicated first in order to be applied before
#' children classifiers.
#' If classifiers is NULL, the method will use all classifiers in database.
#' @param cell_types list of cell types containing models to be used
#' for classification, only applicable if the models have been saved to package.
#' @param path_to_models path to the folder containing the list of models.
#' As default value, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param chunk_size size of data chunks to be predicted separately.
#' This option is recommended for large datasets to reduce running time.
#' Default value at 5000, because smaller datasets can be predicted rapidly.
#' @param ignore_ambiguous_result return all ambiguous predictions
#' (multiple cell types) to empty
#' When this parameter turns to TRUE,
#' most probably predicted cell types will be ignored.
#' @param cluster_slot name of slot in meta data containing cluster
#' information, in case users want to have additional cluster-level
#' prediction
#'
#' @return the input object with new slots in cells meta data
#' New slots are: predicted_cell_type, most_probable_cell_type,
#' slots in form of [cell_type]_p, [cell_type]_class, and clust_pred
#' (if cluster_slot was provided).
#'
#' @importFrom Seurat GetAssayData
#' @import dplyr
#' @importFrom stats predict
#'
#' @rdname internal
classify_cells_seurat <-
  function(classify_obj, classifiers = NULL, cell_types = "all",
           chunk_size = 5000, path_to_models = "default",
           ignore_ambiguous_result = FALSE, cluster_slot,
           seurat_assay, seurat_slot) {
  if (is.null(classifiers)) {
    model_list <- load_models(path_to_models)

    if (length(cell_types) >= 1 && all(cell_types != "all"))
      classifiers <- subset_models(model_list, cell_types)
    else
      classifiers <- model_list
  }

  cell_types <- names(classifiers)
  for (cell_type in cell_types) {
    if (any(!marker_genes(classifiers[[cell_type]]) %in% rownames(classify_obj))) {
      warning('All genes from ', cell_type, ' classifier model must be present in the dataset to perform classification. ',
              'Classification of ', cell_type, ' skipped.\n', call. = FALSE, immediate. = TRUE)
      classifiers <- classifiers[!names(classifiers) %in% cell_type]
    }
  }

  union.marker_genes <- unique(unname(unlist(lapply(classifiers,
                                                    function(x) marker_genes(x)))))
  mat = Seurat::GetAssayData(object = classify_obj,
                             assay = seurat_assay, slot = seurat_slot)
  # if expression matrix is not dgCMatrix: DelayedMatrix for ex.
  if (!is(mat, 'dgCMatrix'))
    mat <- as(mat, "dgCMatrix")

  # reduce marker genes to reduce computational complexity
  mat <- select_marker_genes(mat, union.marker_genes)
  mat <- t(transform_to_zscore(t(as.matrix(mat))))

  nchunks = ceiling(ncol(classify_obj)/chunk_size)
  for (i in seq(1, nchunks)) {
    idx.chunk = seq((i - 1) * chunk_size + 1,
                    min(ncol(classify_obj), i * chunk_size))
    obj.chunk <- classify_obj[, idx.chunk]
    mat.chunk <- mat[, idx.chunk, drop = FALSE]

    # create an empty cell type for all cells
    pred_cells <- c(rep("unknown", ncol(mat.chunk)))
    names(pred_cells) <- colnames(mat.chunk)

    # run predictors
    for (classifier in classifiers) {
      if (!is.na(parent(classifier))) {
        applicable_mat <- verify_parent(mat.chunk, classifier, obj.chunk[[]])
        # no parent classifier provided or no positive to parent classifier
        if (is.null(applicable_mat)) next
      } else applicable_mat <- mat.chunk

      filtered_mat <- select_marker_genes(applicable_mat, marker_genes(classifier))
      filtered_mat <- t(as.matrix(filtered_mat))

      prediction <- make_prediction(
        filtered_mat, classifier, pred_cells, ignore_ambiguous_result
      )
      pred <- prediction$pred
      pred_cells <- prediction$pred_cells

      # add prediction to meta data: each classifier has p & class cols
      for (colname in colnames(pred)) {
        obj.chunk[[colname]] <- pred[, colname, drop = FALSE]
      }
    }

    if (any(pred_cells != "")) {
      pred_cells <- gsub("unknown/", "", pred_cells)

      # ignore ambiguous results
      if (ignore_ambiguous_result == TRUE) {
        pred_cells <- unlist(
          lapply(pred_cells, function(x)
            if (length(unlist(strsplit(x, split = '/'))) <= 1) {x}
            else {'ambiguous'})
        )
      }

      # add cell type to meta data
      obj.chunk[['predicted_cell_type']] <- pred_cells

      # simplify result can only happen when not ignore ambiguous results
      if (ignore_ambiguous_result == FALSE)
        obj.chunk[['most_probable_cell_type']] <- simplify_prediction(
          obj.chunk[[]], obj.chunk$predicted_cell_type, classifiers)
    }

    if (i == 1) classified_obj <- obj.chunk
    else classified_obj <- merge(classified_obj, obj.chunk)
  }

  if (!is.null(cluster_slot)
      & cluster_slot %in% colnames(classified_obj[[]])
      & !ignore_ambiguous_result) {
    clusts <- as.factor(classified_obj[[]][, cluster_slot])
    classified_obj$clust_pred <-
      classify_clust(clusts, classified_obj$most_probable_cell_type)
  }

  new.cols <- colnames(classified_obj[[]])[!colnames(classified_obj[[]])
                                           %in% colnames(classify_obj[[]])]
  for (colname in new.cols)
    classify_obj[[colname]] <- classified_obj[[colname]]

  return(classify_obj)
}

#' Classify cells from multiple models for SCE object as input
#'
#' @param classify_obj the SCE object containing cells to be classified
#' @param sce_assay name of assay to use in SCE object
#' @param classifiers list of classification models.
#' The model is obtained from train_classifier function or available in current
#' working space.
#' Users may test the model using test_classifier before using this function.
#' If classifiers contain classifiers for sub cell types, classifiers for
#' parent cell type must be indicated first in order to be applied before
#' children classifiers.
#' If classifiers is NULL, the method will use all classifiers in database.
#' @param cell_types list of cell types containing models to be used
#' for classification, only applicable if the models have been saved to package.
#' @param path_to_models path to the folder containing the list of models.
#' As default value, the pretrained models in the package will be used.
#' If user has trained new models, indicate the folder containing the
#' new_models.rda file.
#' @param chunk_size size of data chunks to be predicted separately.
#' This option is recommended for large datasets to reduce running time.
#' Default value at 5000, because smaller datasets can be predicted rapidly.
#' @param ignore_ambiguous_result return all ambiguous predictions
#' (multiple cell types) to empty
#' When this parameter turns to TRUE,
#' most probably predicted cell types will be ignored.
#' @param cluster_slot name of slot in meta data containing cluster
#' information, in case users want to have additional cluster-level
#' prediction
#'
#' @return the input object with new slots in cells meta data
#' New slots are: predicted_cell_type, most_probable_cell_type,
#' slots in form of [cell_type]_p, [cell_type]_class, and clust_pred
#' (if cluster_slot was provided).
#'
#' @import SingleCellExperiment
#' @importFrom SummarizedExperiment assay colData
#'
#' @rdname internal
classify_cells_sce <-
  function(classify_obj, classifiers = NULL, cell_types = "all",
           chunk_size = 5000, path_to_models = "default",
           ignore_ambiguous_result = FALSE, sce_assay, cluster_slot = NULL) {
  # solve duplication of cell names
  colnames(classify_obj) <- make.unique(colnames(classify_obj), sep = '_')

  if (is.null(classifiers)) {
    model_list <- load_models(path_to_models)

    if (cell_types >=1 && cell_types != "all") 
      classifiers <- subset_models(model_list, cell_types)
    else 
      classifiers <- model_list
  }

  cell_types <- names(classifiers)
  for (cell_type in cell_types) {
    if (any(!marker_genes(classifiers[[cell_type]]) %in% rownames(classify_obj))) {
      warning('Feature genes for ', cell_type, ' is not available in the input SCE object. ',
              'Classification of ', cell_type, ' skipped.\n', call. = FALSE, immediate. = TRUE)
      classifiers <- classifiers[!names(classifiers) %in% cell_type]
    }
  }

  union.marker_genes <- unique(unname(unlist(lapply(classifiers,
                                                function(x) marker_genes(x)))))
  # if expression matrix is not dgCMatrix: DelayedMatrix for ex.
  mat = SummarizedExperiment::assay(classify_obj, sce_assay, withDimnames = FALSE)
  if (!is(mat, 'dgCMatrix'))
    mat <- as(mat, "dgCMatrix")

  # reduce marker_genes to reduce computational complexity
  mat <- select_marker_genes(mat, union.marker_genes)
  mat <- t(transform_to_zscore(t(as.matrix(mat))))

  # split dataset into multiple chunks to reduce running time
  nchunks = ceiling(ncol(classify_obj)/chunk_size)
  for (i in seq(1, nchunks)) {
    idx.chunk = seq((i - 1) * chunk_size + 1,
                    min(ncol(classify_obj), i * chunk_size))
    obj.chunk <- classify_obj[, idx.chunk]
    mat.chunk <- mat[, idx.chunk, drop = FALSE]

    # create an empty cell type for all cells
    pred_cells <- c(rep("unknown", ncol(mat.chunk)))
    names(pred_cells) <- colnames(mat.chunk)

    # run predictors
    for (classifier in classifiers) {
      if (!is.na(parent(classifier))) {
        applicable_mat <- verify_parent(mat.chunk, classifier,
                                        SummarizedExperiment::colData(obj.chunk))
        # no parent classifier provided or no positive to parent classifier
        if (is.null(applicable_mat)) next
      } else applicable_mat <- mat.chunk

      filtered_mat <- select_marker_genes(applicable_mat, marker_genes(classifier))
      filtered_mat <- t(as.matrix(filtered_mat))

      prediction <- make_prediction(
        filtered_mat, classifier, pred_cells, ignore_ambiguous_result)

      pred <- prediction$pred
      pred_cells <- prediction$pred_cells
      # add prediction to meta data: 2 cols: p, class
      for (colname in colnames(pred)) {
        obj.chunk[[colname]] <- unlist(
          lapply(colnames(obj.chunk), function(x)
            if (x %in% rownames(pred)) {pred[x, colname]}
            else {NA})
        )
      }
    }

    if (any(pred_cells != "")) {
      pred_cells <- gsub("unknown/", "", pred_cells)
      # double check if there is more than one predicted cell type
      if (ignore_ambiguous_result == TRUE) {
        pred_cells <- unlist(
          lapply(pred_cells, function(x)
            if (length(unlist(strsplit(x, split = '/'))) <= 1) {x}
            else {'ambiguous'})
          )
      }

      # add cell type to meta data
      obj.chunk$predicted_cell_type <- pred_cells
      # this will be ignored if ignore ambiguous result is on
      if (ignore_ambiguous_result == FALSE)
        obj.chunk$most_probable_cell_type <- simplify_prediction(
          as.matrix(SummarizedExperiment::colData(obj.chunk)),
          obj.chunk$predicted_cell_type, classifiers
        )
    }

    if (i == 1) classified_obj <- obj.chunk
    else classified_obj <- cbind(classified_obj, obj.chunk)
  }

  if (!is.null(cluster_slot)
      & cluster_slot %in% colnames(SummarizedExperiment::colData(classified_obj))
      & !ignore_ambiguous_result) {
    clusts <- SummarizedExperiment::colData(classified_obj)[, cluster_slot]
    classified_obj$clust_pred <-
      classify_clust(clusts, classified_obj$most_probable_cell_type)
  }

  return(classified_obj)
}
